#pragma once
#include <iostream>
#include <string>
#include "Socket.hpp"
#include <memory>
#include <jsoncpp/json/json.h>
#include "Log.hpp"
#include <functional>
// 实现一个自定义的网络版的计算器
// tcp是面向字节流的
// 当读取的时候，可能读到一个完整的json请求，也可能读到半个？
// read->并不能保证读取到的报文的完整性！它只保证如果有数据就读上来。
// 当我们tcp中读取数据的时候，读取到的报文不完整，或者多读了，导致下一个报文不完整，这个问题叫做“粘包”问题！
// 所以需要自己实现协议的解析和序列化

// content_len jsonstring
// 50\r\n{"x":1,"y":2,"oper":"+"}\r\n
// 50
// {"x":1,"y":2,"oper":"+"}
using namespace SocketModule;
using namespace LogModule;

class Request
{
public:
    Request() {}
    Request(int x, int y, char oper) : _x(x), _y(y), _oper(oper)
    {
    }

    std::string Serialize()
    {
        // json做序列化和反序列化处理
        Json::Value root;
        root["x"] = _x;
        root["y"] = _y;
        root["oper"] = _oper;

        Json::FastWriter writer;
        std::string s = writer.write(root);
        return s;
    }

    bool Deserialize(std::string &in)
    {
        Json::Value root;
        Json::Reader reader;
        bool ok = reader.parse(in, root);
        if (ok)
        {
            _x = root["x"].asInt();
            _y = root["y"].asInt();
            _oper = root["oper"].asInt(); // ?
        }
        return ok;
    }
    ~Request() {}
    int X() { return _x; }
    int Y() { return _y; }
    char Oper() { return _oper; }

private:
    int _x;
    int _y;
    char _oper;
};

class Response
{
public:
    Response() {}
    Response(int result, int code = 0) : _result(result), _code(code) {}
    std::string Serialize()
    {
        // json做序列化处理
        Json::Value root;
        root["result"] = _result;
        root["code"] = _code;

        Json::FastWriter writer;
        std::string s = writer.write(root);
        return s;
    }

    bool DeSerialize(std::string &in)
    {
        // json做反序列化处理
        Json::Value root;
        Json::Reader reader;
        bool ok = reader.parse(in, root);
        if (ok)
        {
            _result = root["result"].asInt();
            _code = root["code"].asInt();
        }
        return ok;
    }
    ~Response() {}
    void SetResult(int result) { _result = result; }
    void SetCode(int code) { _code = code; }
    void ShowResult()
    {
        std::cout << "result:" << _result << "[" << _code << "]" << std::endl;
    }

private:
    int _result; // 运算结果
    int _code;   // 0表示成功，1,2,3,4->不同的运算异常
};

const std::string sep = "\r\n";

using func_t = std::function<Response(Request &)>;

// 实现一个自定义的网络版的计算器协议
//
class Protocol
{
public:
    Protocol() {}
    Protocol(func_t func) : _func(func)
    {
    }
    std::string Encode(const std::string jsonstr)
    {
        std::string len = std::to_string(jsonstr.size());
        std::string package = len + sep + jsonstr + sep;
        return package;
    }

    // 1. 判断报文的完整性
    // 2. 如果包含至少一个完整请求，提取并移出它.
    bool Decode(std::string &buffer, std::string *package)
    {
        // 解析报文长度
        ssize_t pos = buffer.find(sep);
        if (pos == std::string::npos)
            return false;
        std::string package_len_str = buffer.substr(0, pos);
        int package_len_int = std::stoi(package_len_str);

        int target_len = package_len_str.size() + package_len_int + 2 * sep.size();
        if (buffer.size() < target_len)
            return false;

        // 解析报文内容
        *package = buffer.substr(pos + sep.size(), package_len_int);
        buffer.erase(0, target_len);

        return true;
    }
    void GetRequest(std::shared_ptr<Socket> &sock, InetAddr &client)
    {
        // 读取
        std::string buffer_queue;
        while (true)
        {
            int n = sock->Recv(&buffer_queue);
            if (n > 0)
            {
                // 1. 解析报文，提取完整的json请求，如果不完整就让服务器继续读取
                std::string json_package;
                while (Decode(buffer_queue, &json_package))
                {
                    // 2. 反序列化
                    // 2.
                    // {"x":1,"y":2,"oper":"+"}
                    Request req;
                    bool ok = req.Deserialize(json_package);
                    if (!ok)
                        continue;

                    Response resp = _func(req);

                    // 序列化
                    std::string json_str = resp.Serialize();

                    // 添加自定义长度
                    std::string send_str = Encode(json_str);

                    // 直接发送
                    sock->Send(send_str);
                }
            }
            else if (n == 0)
            {
                LOG(LogLevel::INFO) << "client:" << client.StringAddr() << "Quit";
                break;
            }
            else
            {
                LOG(LogLevel::WARNING) << "client:" << client.StringAddr() << ", Recv error";
                break;
            }
        }
        sock->Close();
    }

    bool GetResponse(std::shared_ptr<Socket> &client, std::string &resp_str, Response *resp)
    {
        // 面向字节流的
        while (true)
        {
            int n = client->Recv(&resp_str);
            if (n > 0)
            {
                std::cout << "----------------resp_buffer-----------------" << std::endl;
                std::cout << resp_str << std::endl;
                std::cout << "--------------------------------------------" << std::endl;
                std::string json_package;
                // bool ret = Decode(resp_str, &json_package);
                // if (!ret)
                //     continue;

                while (Decode(resp_str, &json_package))
                {
                    std::cout << "----------------json_package---------------" << std::endl;
                    std::cout << json_package << std::endl;
                    std::cout << "-------------------------------------------" << std::endl;
                    // 这到这能保证一定拿到了一个完整的json请求
                    // 3. 反序列化
                    resp->DeSerialize(json_package);
                    resp->ShowResult();
                }
                return true;
            }
            else if (n == 0)
            {
                std::cout << "server quit" << std::endl;
                return false;
            }
            else
            {
                std::cout << "Recv error" << std::endl;
                return false;
            }
        }
    }

    std::string BuildRequestString(int x, int y, char oper)
    {
        // 1. 构建一个完整的请求
        Request req(x, y, oper);
        std::string json_str = req.Serialize();

        // debug
        std::cout << "----------------json_str---------------" << std::endl;
        std::cout << json_str << std::endl;
        std::cout << "---------------------------------------" << std::endl;

        // 2. 添加自定义长度
        return Encode(json_str);
    }

    ~Protocol() {}

private:
    // Response _req;
    // Response _resp;
    func_t _func;
};